data structure with python